package com.hmdp.utils;

import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import com.hmdp.dto.Result;
import com.hmdp.entity.Shop;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.util.concurrent.TimeUnit;
import java.util.function.Function;

@Slf4j
@Component
public class CacheClient {

    private final StringRedisTemplate stringRedisTemplate;

    public CacheClient(StringRedisTemplate stringRedisTemplate) {
        this.stringRedisTemplate = stringRedisTemplate;
    }

    /**
     * 将Redis的key的值设置为value 并添加过期时间TTL
     */
    public void set(String key, Object value, Long time, TimeUnit timeUnit){
        stringRedisTemplate.opsForValue().set(key, JSONUtil.toJsonStr(value),time,timeUnit);
    }

    /**
     * 将Redis的key的值设置为value 并添加逻辑过期时间
     */
    public void setExpireTime(String key, Object value, Long time, TimeUnit timeUnit){
        RedisData redisData = new RedisData();
        redisData.setData(value);
        redisData.setExpireTime(LocalDateTime.now().plusSeconds(timeUnit.toSeconds(time)));

        stringRedisTemplate.opsForValue().set(key, JSONUtil.toJsonStr(redisData));
    }

    /**
     * 空对象解决缓存穿透
     * @param keyPrefix key前缀
     * @param id id
     * @param type 返回类型
     * @param dbCallBack 函数接口执行sql
     * @param time 时间
     * @param timeUnit 时间单位
     */
    public <R,ID> R cachePenetration(String keyPrefix, ID id, Class<R> type, Function<ID,R> dbCallBack,Long time,TimeUnit timeUnit) {
        String key = keyPrefix + id;

        // 先从redis读取数据
        String shopJson = stringRedisTemplate.opsForValue().get(key);

        // 判断是否存在
        if (StrUtil.isNotBlank(shopJson)) { // 存在就直接用缓存的
            return JSONUtil.toBean(shopJson, type);
        }

        // 判断是否null 就是空值 用于缓存与数据库都不存在这个数据时 让请求不进入数据库中 减轻数据库压力
        if (shopJson != null) {
            return null;
        }

        // 不存在就查数据库
        R res = dbCallBack.apply(id);

        if (res == null) {
            // 将空值写入redis 解决缓存穿透
            set(key, "", RedisConstants.CACHE_NULL_TTL, TimeUnit.MINUTES);
            return null;
        }

        // 将数据库的数据存入redis
        set(key,res,time,timeUnit);
        return res;
    }
}
